package main

import (
	"fmt"
	"os"
)

func menu() {
	fmt.Println("欢迎来到学员管理系统")
	fmt.Println("1:添加学员")
	fmt.Println("2:修改学员")
	fmt.Println("3:展示学员")
	fmt.Println("4:退出系统")

}

//获取用户输入的学员信息
func getInput() *student {
	var (
		id    int
		name  string
		class string
	)
	fmt.Println("请按照要求输入学员信息")
	fmt.Print("请按照要求输入学员id:")
	fmt.Scanf("%d\n", &id)
	fmt.Print("请按照要求输入学员姓名:")
	fmt.Scanf("%s\n", &name)
	fmt.Print("请按照要求输入学员班级:")
	fmt.Scanf("%s\n", &class)
	stu := newStu(id, name, class)
	return stu

}

func main() {

	sm := newStuMgr()
	for {
		//打印菜单
		menu()
		//等待用户输入
		var input int
		fmt.Scanf("%d\n", &input)

		switch input {
		case 1:
			stu := getInput()
			sm.add(stu)
		case 2:
			stu := getInput()
			sm.edit(stu)
		case 3:
			sm.show()
		case 4:
			os.Exit(0)
		default:
			return
		}
	}

}
